from histogram import HistogramCollection, Histogram1D, Histogram2D

hcoll = HistogramCollection(
    [Histogram1D(name='m_pipipi',
                 nbins=1010,
                 arange=(0.4, 5.5),
                 xlabel=r'M($\pi_s\pi\pi$)',
                 xunits=r'GeV',
                 assoc_cols='m_pipipi'),
     Histogram1D(name='m_kkk',
                 nbins=1000,
                 arange=(1.5, 6.5),
                 xlabel=r'M($K_sKK$)',
                 xunits=r'GeV',
                 assoc_cols='m_kkk'),
     Histogram1D(name='m_pikpi',
                 nbins=1000,
                 arange=(0.7, 5.7),
                 xlabel=r'M($\pi_sK^+\pi^-$)',
                 xunits=r'GeV',
                 assoc_cols='m_pikpi'),
     Histogram1D(name='m_pipiK',
                 nbins=1000,
                 arange=(0.7, 5.7),
                 xlabel=r'M($\pi_sK^-\pi^+ $)',
                 xunits=r'GeV',
                 assoc_cols='m_pipik'),
     Histogram1D(name='q_kpi',
                 nbins=200,
                 arange=(0,0.02),
                 xlabel=r'M($\pi_sK^+\pi^-$)-M($K^+\pi^-$)-M$_\pi$',
                 xunits='GeV',
                 title='$Q$ for $K^+\pi^-$',
                 assoc_cols='q_kpi',
                 ),
     Histogram1D(name='q_pik',
                 nbins=200,
                 arange=(0,0.02),
                 xlabel=r'M($\pi_sK^-\pi^+$)-M($K^-\pi^+$)-M$_\pi$',
                 xunits='GeV',
                 title='$Q$ for $K^-\pi^+$',
                 assoc_cols='q_pik',
                 ),
     Histogram1D(name='pcm_pikpi',
                 nbins=1250,
                 arange=(1,7),
                 xlabel=r'$|p^*|_{D^*} (D^*\to\pi_sK^+\pi^-)$',
                 xunits='GeV',
                 assoc_cols='pcm_pikpi',
                 ),
     Histogram1D(name='pcm_pipik',
                 nbins=1250,
                 arange=(1,7),
                 xlabel=r'$|p^*|_{D^*} (D^*\to\pi_sK^-\pi^+)$',
                 xunits='GeV',
                 assoc_cols='pcm_pipik',
                 ),
     Histogram1D(name='kpi_theta_star',
                 nbins=1000,
                 arange=(-1,1),
                 xlabel=r'$\theta^*$ ($K^+\pi^-$ system)',
                 xunits='GeV',
                 assoc_cols='kpi_theta_star',
                 ),
     Histogram1D(name='pik_theta_star',
                 nbins=1000,
                 arange=(-1,1),
                 xlabel=r'$\theta^*$ ($K^-\pi^+$ system)',
                 xunits='GeV',
                 assoc_cols='pik_theta_star',
                 ),
     ])


